/**
 * @Question.Title: 串联所有单词的子串
 * @Question.No: 30
 * @Author: Dang Qi
 * @Date: 2021-04-25 21:45:15
 * @Description: 
 */
//给定一个字符串 s 和一些长度相同的单词 words。找出 s 中恰好可以由 words 中所有单词串联形成的子串的起始位置。 
//
// 注意子串要与 words 中的单词完全匹配，中间不能有其他字符，但不需要考虑 words 中单词串联的顺序。 
//
// 
//
// 示例 1： 
//
// 输入：
//  s = "barfoothefoobarman",
//  words = ["foo","bar"]
//输出：[0,9]
//解释：
//从索引 0 和 9 开始的子串分别是 "barfoo" 和 "foobar" 。
//输出的顺序不重要, [9,0] 也是有效答案。
// 
//
// 示例 2： 
//
// 输入：
//  s = "wordgoodgoodgoodbestword",
//  words = ["word","good","best","word"]
//输出：[]
// 
// Related Topics 哈希表 双指针 字符串 
// 👍 461 👎 0

package dq.leetcode.editor.cn;

import java.util.*;

public class SubstringWithConcatenationOfAllWords {
    public static void main(String[] args) {
        Solution solution = new SubstringWithConcatenationOfAllWords().new Solution();
        String s = "wordgoodgoodgoodbestword";
        String[] words = {"word","good","best","good"};
        List<Integer> substring = solution.findSubstring(s, words);
        substring.forEach(System.out::println);
    }
        //leetcode submit region begin(Prohibit modification and deletion)
class Solution {
    public List<Integer> findSubstring(String s, String[] words) {
        List<Integer> res = new ArrayList<>();
        //字典保存单词
        Map<String, Integer> dict = new HashMap<>();
        Arrays.stream(words).forEach(str -> dict.put(str, dict.getOrDefault(str, 0) + 1));
        //每一个单词的长度
        int len = words[0].length();
        int end = s.length() - len * words.length;
        for(int i = 0; i <= end; i++){
            if(judge(len, i, (words.length - 1) * len + i, new HashMap<>(dict), s)){
                res.add(i);
            }
        }
        return res;
    }
    public boolean judge(int len, int start, int end, Map<String, Integer> dict, String s){

        for(int i = start; i <= end; i+=len){
            String str = s.substring(i, i+len);
            Integer num = dict.get(str);
            if(num == null) return false;
            if(num == 1){
                dict.remove(str);
            }else{
                dict.put(str, num-1);
            }
        }
        return dict.isEmpty();
    }
}
//leetcode submit region end(Prohibit modification and deletion)

}